if weights_only:
    try:
         return _load(
                  opened_zipfile,
                  map_location,
                  _weights_only_unpickler,
                  overall_storage=overall_storage,
                  **pickle_load_args,
              )
except pickle.UnpicklingError as e:
>   raise pickle.UnpicklingError(_get_wo_message(str(e))) from None
E    pickle.UnpicklingError: Weights only load failed. This file can still be loaded, to do so you have two options, 
[do those steps only if you trust the source of the checkpoint]. 
E    (1) Re-running `torch.load` with `weights_only` set to `False` will likely succeed, but it can result in arbitrary 
code execution. Do it only if you got the file from a trusted source.
E    (2) Alternatively, to load with `weights_only=True` please check the recommended steps in the following 
error message.
E    WeightsUnpickler error: Unsupported global: GLOBAL dgl.graphbolt.impl.fused_csc_sampling_graph.FusedCSCSamplingGraph 
was not an allowed global by default.  Please use `torch.serialization.add_safe_globals([FusedCSCSamplingGraph])` 
or the `torch.serialization.safe_globals([FusedCSCSamplingGraph])` context manager to allow list this global if you 
trust this class/function.
E                           
E    Check the documentation of torch.load to learn more about types accepted by default with 
weights_only https://pytorch.org/docs/stable/generated/torch.load.html.

/usr/local/lib/python3.12/dist-packages/torch/serialization.py:1455: UnpicklingError